Skip to content

[TRTLLM-12154][test] Add Qwen3-32B FP8 disagg stress test#14278

Merged
tburt-nv merged 4 commits into
NVIDIA:mainfrom
brnguyen2:brnguyen/qwen3-32b-fp8-disagg-stress
Jun 11, 2026
Merged

[TRTLLM-12154][test] Add Qwen3-32B FP8 disagg stress test#14278
tburt-nv merged 4 commits into
NVIDIA:mainfrom
brnguyen2:brnguyen/qwen3-32b-fp8-disagg-stress

Conversation

@brnguyen2

@brnguyen2 brnguyen2 commented May 18, 2026

Copy link
Copy Markdown
Collaborator

Summary

Adds Qwen3-32B FP8 disaggregated serving coverage for the Eagle3 path on 8 GPUs.

The new disagg config uses:

  • target model: Qwen3/Qwen3-32B-FP8
  • Eagle3 draft model: Zhi-Create-Qwen3-32B-Eagle3, resolved relative to llm_models_root()
  • 4 context workers with TP1 and 1 generation worker with TP4
  • FP8 KV cache, chunked prefill, block/partial reuse, and DEFAULT cache transceiver
  • top-level speculative_config so context and generation workers agree on the cache-state handshake

The disagg harness now resolves relative model and speculative_config.speculative_model values through llm_models_root() while preserving absolute paths. This keeps the YAML portable across CI/local model roots.

Also makes the GSM8K lm-eval parser tolerate padded table cells, matching the output shape emitted by the local-completions run.

Test entries:

  • test_disaggregated_qwen3_32b_fp8 smoke test in l0_dgx_h200.yml
  • test_disaggregated_stress_test::qwen3_32b_fp8_stress in qa/llm_function_stress.txt, request_count=10000, accuracy_threshold=0.42

Refs: TRTLLM-12154

Test plan

  • Targeted smoke test passed before the upstream rebase with FP8 KV + Eagle3
  • Rebuild on upstream main after rebase
  • Rerun tests/integration/defs/disaggregated/test_disaggregated.py::test_disaggregated_qwen3_32b_fp8 after rebuild
  • Locally validated the stress node with temporary request/output/lm-eval limits; aiperf completed 16/16 requests and GSM8K parsed as 0.75 against the 0.42 threshold. Temporary limits are not committed.
  • Pre-merge CI

Summary by CodeRabbit

Release Notes

  • Tests
    • Added disaggregated test coverage for Qwen3-32B-FP8 model with speculative decoding configuration.
    • Improved GSM8K accuracy extraction to support variable whitespace and both integer and decimal values.
    • Extended stress test parametrization to include new Qwen3-32B-FP8 test scenarios on H200 GPUs.

Review Change Stack

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 181a22d to ce5f6a0 Compare May 19, 2026 13:07
@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot help

@github-actions

Copy link
Copy Markdown

GitHub Bot Help

/bot [-h] ['run', 'kill', 'skip', 'reuse-pipeline'] ...

Provide a user friendly way for developers to interact with a Jenkins server.

Run /bot [-h|--help] to print this help message.

See details below for each supported subcommand.

Details

run [--reuse-test (optional)pipeline-id --disable-fail-fast --skip-test --stage-list "A10-PyTorch-1, xxx" --gpu-type "A30, H100_PCIe" --test-backend "pytorch, cpp" --add-multi-gpu-test --only-multi-gpu-test --disable-multi-gpu-test --post-merge --extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx" --detailed-log --debug(experimental) --high-priority]

Launch build/test pipelines. All previously running jobs will be killed.

--reuse-test (optional)pipeline-id (OPTIONAL) : Allow the new pipeline to reuse build artifacts and skip successful test stages from a specified pipeline or the last pipeline if no pipeline-id is indicated. If the Git commit ID has changed, this option will be always ignored. The DEFAULT behavior of the bot is to reuse build artifacts and successful test results from the last pipeline.

--disable-reuse-test (OPTIONAL) : Explicitly prevent the pipeline from reusing build artifacts and skipping successful test stages from a previous pipeline. Ensure that all builds and tests are run regardless of previous successes.

--disable-fail-fast (OPTIONAL) : Disable fail fast on build/tests/infra failures.

--skip-test (OPTIONAL) : Skip all test stages, but still run build stages, package stages and sanity check stages. Note: Does NOT update GitHub check status.

--stage-list "A10-PyTorch-1, xxx" (OPTIONAL) : Only run the specified test stages. Supports wildcard * for pattern matching (e.g., "*PerfSanity*" matches all stages containing PerfSanity). Examples: "A10-PyTorch-1, xxx", "PerfSanity". Note: Does NOT update GitHub check status.

--gpu-type "A30, H100_PCIe" (OPTIONAL) : Only run the test stages on the specified GPU types. Examples: "A30, H100_PCIe". Note: Does NOT update GitHub check status.

--test-backend "pytorch, cpp" (OPTIONAL) : Skip test stages which don't match the specified backends. Only support [pytorch, cpp, tensorrt, triton]. Examples: "pytorch, cpp" (does not run test stages with tensorrt or triton backend). Note: Does NOT update GitHub pipeline status.

--only-multi-gpu-test (OPTIONAL) : Only run the multi-GPU tests. Note: Does NOT update GitHub check status.

--disable-multi-gpu-test (OPTIONAL) : Disable the multi-GPU tests. Note: Does NOT update GitHub check status.

--add-multi-gpu-test (OPTIONAL) : Force run the multi-GPU tests in addition to running L0 pre-merge pipeline.

--post-merge (OPTIONAL) : Run the L0 post-merge pipeline instead of the ordinary L0 pre-merge pipeline.

--extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx" (OPTIONAL) : Run the ordinary L0 pre-merge pipeline and specified test stages. Supports wildcard * for pattern matching. Examples: --extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx", --extra-stage "Post-Merge".

--detailed-log (OPTIONAL) : Enable flushing out all logs to the Jenkins console. This will significantly increase the log volume and may slow down the job.

--debug (OPTIONAL) : Experimental feature. Enable access to the CI container for debugging purpose. Note: Specify exactly one stage in the stage-list parameter to access the appropriate container environment. Note: Does NOT update GitHub check status.

--high-priority (OPTIONAL) : Run the pipeline with high priority. This option is restricted to authorized users only and will route the job to a high-priority queue.

kill

kill

Kill all running builds associated with pull request.

skip

skip --comment COMMENT

Skip testing for latest commit on pull request. --comment "Reason for skipping build/test" is required. IMPORTANT NOTE: This is dangerous since lack of user care and validation can cause top of tree to break.

reuse-pipeline

reuse-pipeline

Reuse a previous pipeline to validate current commit. This action will also kill all currently running builds associated with the pull request. IMPORTANT NOTE: This is dangerous since lack of user care and validation can cause top of tree to break.

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49195 [ run ] triggered by Bot. Commit: ce5f6a0 Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49195 [ run ] completed with state SUCCESS. Commit: ce5f6a0
/LLM/main/L0_MergeRequest_PR pipeline #38871 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch 5 times, most recently from c7357ef to e4166bb Compare May 20, 2026 18:55
@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49487 [ run ] triggered by Bot. Commit: e4166bb Link to invocation

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from e4166bb to 690a201 Compare May 20, 2026 21:10
@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot kill

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast --stage-list "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49493 [ run ] triggered by Bot. Commit: 690a201 Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49494 [ kill ] triggered by Bot. Commit: 690a201 Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49493 [ run ] completed with state ABORTED. Commit: 690a201

Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49487 [ run ] completed with state ABORTED. Commit: e4166bb

Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49494 [ kill ] completed with state SUCCESS. Commit: 690a201
Successfully killed previous jobs for commit 690a201

Link to invocation

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast --stage-list "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49495 [ run ] triggered by Bot. Commit: 690a201 Link to invocation

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

Verified that the targeted rerun reached the newly added test and it passed:

DGX_H200-8_GPUs-PyTorch-Post-Merge-1/disaggregated/test_disaggregated.py::test_disaggregated_qwen3_32b_fp8[Qwen3/Qwen3-32B-FP8]

The selected test run completed successfully with this pytest summary: 16 passed, 12 skipped, 17175 deselected.

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 690a201 to 2c22fef Compare May 21, 2026 00:22
@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49495 [ run ] completed with state SUCCESS. Commit: 690a201
/LLM/main/L0_MergeRequest_PR pipeline #39133 (Partly Tested) completed with status: 'SUCCESS'

CI Report

Link to invocation

@brnguyen2 brnguyen2 marked this pull request as ready for review May 21, 2026 00:28
@brnguyen2 brnguyen2 requested a review from a team as a code owner May 21, 2026 00:28
@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49935 [ run ] triggered by Bot. Commit: 02a618a Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #49935 [ run ] completed with state SUCCESS. Commit: 02a618a
/LLM/main/L0_MergeRequest_PR pipeline #39508 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 02a618a to 6f5e8c2 Compare June 8, 2026 21:52
@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #52863 [ run ] triggered by Bot. Commit: 6f5e8c2 Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #52863 [ run ] completed with state SUCCESS. Commit: 6f5e8c2
/LLM/main/L0_MergeRequest_PR pipeline #42115 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 6f5e8c2 to 50561e0 Compare June 9, 2026 21:45
@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot kill

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53152 [ run ] triggered by Bot. Commit: 50561e0 Link to invocation

@brnguyen2

Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53154 [ kill ] triggered by Bot. Commit: 50561e0 Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53152 [ run ] completed with state ABORTED. Commit: 50561e0

Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53154 [ kill ] completed with state SUCCESS. Commit: 50561e0
Successfully killed previous jobs for commit 50561e0

Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53156 [ run ] triggered by Bot. Commit: 50561e0 Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53156 [ run ] completed with state FAILURE. Commit: 50561e0
/LLM/main/L0_MergeRequest_PR pipeline #42359 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@tburt-nv

Copy link
Copy Markdown
Collaborator

/bot skip --comment "DGX_B200-8_GPUs-PyTorch-2 and RTXPro6000D-PyTorch-1 passed in 42359"

Add a Qwen3-32B FP8 disaggregated serving smoke and stress test that exercises Eagle3 with 4x TP1 context workers and 1x TP4 generation worker on 8 GPUs.

The YAML enables FP8 KV cache, chunked prefill, block and partial reuse, cache transfer, and a top-level Eagle3 speculative_config shared by context and generation workers. The draft model is stored as a model-root-relative path and the disagg harness now resolves relative model and speculative_model values through llm_models_root while preserving absolute paths.

Wire the smoke test into the H200 L0 list and the full 10k-request stress case into the QA stress list. Add Qwen-specific output substring checks and keep the stress accuracy threshold aligned with the adjacent GPT-OSS stress case.

Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
…t_bf16_mtp[mtp_on]

The test crashes during autotuner warmup with
'NoneType' object has no attribute 'gather_ids' at
modeling_speculative.py:1748 when MTP eagle one-model is combined
with Qwen3.5-35B-A3B. Pre-existing regression on main introduced by
the Qwen3.5 VL MoE landing (96a4a09); unrelated to this PR's
changes. Tracked in https://nvbugs/6206179.

Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 50561e0 to 0bedcb2 Compare June 11, 2026 21:57
@tburt-nv

Copy link
Copy Markdown
Collaborator

/bot skip --comment "changes are identical"

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53710 [ skip ] triggered by Bot. Commit: 0bedcb2 Link to invocation

@tensorrt-cicd

Copy link
Copy Markdown
Collaborator

PR_Github #53710 [ skip ] completed with state SUCCESS. Commit: 0bedcb2
Skipping testing for commit 0bedcb2

Link to invocation

@tburt-nv tburt-nv merged commit ccc0708 into NVIDIA:main Jun 11, 2026
7 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants